#ifndef COOLPOOL_SIMULATION_GRAVITYRIGIDBODYSIMULATION_H_
#define COOLPOOL_SIMULATION_GRAVITYRIGIDBODYSIMULATION_H_

#include "RigidBodySimulation.h"

namespace coolpool
{
	// Rigid body simulation with gravity.
	class GravityRigidBodySimulation : public RigidBodySimulation
	{
	public:

		GravityRigidBodySimulation(const std::vector<Entity *> & entities, int num_workers, const math_tools::Vector3D & gravity);

		virtual ~GravityRigidBodySimulation();

		// Overrides:
		virtual void beforeNextStep();

	private:

		math_tools::Vector3D gravity_;
	};
}

#endif
